package main

import "fmt"

/**
给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。



示例 1:

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:

输入: nums = [1], k = 1
输出: [1]


提示：

1 <= nums.length <= 105
k 的取值范围是 [1, 数组中不相同的元素的个数]
题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的


进阶：你所设计算法的时间复杂度 必须 优于 O(n log n) ，其中 n 是数组大小。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/top-k-frequent-elements
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
*/
func topKFrequent(nums []int, k int) []int {
	// 定义中间结果
	tmpMap := make(map[int]int)
	for _, num := range nums {
		tmpMap[num] += 1
	}
	// 定义数组
	tmpArr := make([][]int, len(nums)+1)
	for key, val := range tmpMap {
		arr := tmpArr[val]
		if arr == nil {
			arr = []int{key}
		} else {
			arr = append(arr, key)
		}
		tmpArr[val] = arr
	}
	// 定义结果
	res := []int{}
	for i := len(nums); i >= 0; i-- {
		arrVal := tmpArr[i]
		if arrVal != nil {
			for _, item := range arrVal {
				res = append(res, item)
			}
		}
	}
	return res[:k]
}
func main() {
	fmt.Println(topKFrequent([]int{1, 1, 1, 2, 2, 3}, 2))
}
